import torch

print(f"PyTorch 版本: {torch.__version__}")
print(f"CUDA 是否可用: {'是' if torch.cuda.is_available() else '否'}")
print(f"GPU 数量: {torch.cuda.device_count()}")
print(f"当前 GPU: {torch.cuda.get_device_name(0)}")
print(f"CUDA 版本: {torch.version.cuda}")